package com.ruben.simplescaffold.filter;

import cn.hutool.core.lang.Opt;
import cn.hutool.core.text.CharSequenceUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import javax.servlet.*;
import javax.servlet.annotation.WebFilter;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;

/**
 * 用户浏览器标识过滤器
 *
 * @author <achao1441470436@gmail.com>
 * @since 2021/8/4 17:09
 */
@Slf4j
@Component
@WebFilter(filterName = "LogFilter", urlPatterns = "/**")
public class UserAgentFilter implements Filter {

    /**
     * 移动端浏览器标识
     */
    private final static List<String> MOBILE_AGENT_LIST = Arrays.asList("Android", "iPhone", "iPod", "iPad", "Windows Phone", "MQQBrowser");
    public static final String FAVICON = "favicon";
    public static final String CONTENT_TYPE = "content-type";
    public static final String FORM_URLENCODED = "application/x-www-form-urlencoded";
    public static final String APPLICATION_JSON = "application/json";
    public static final String USER_AGENT = "User-Agent";
    public static final String WINDOWS_NT = "Windows NT";
    public static final String MSIE = "compatible; MSIE 9.0;";
    public static final String MACINTOSH = "Macintosh";
    public static final String MOBILE_PATH = "/h5";

    /**
     * The <code>doFilter</code> method of the Filter is called by the container
     * each time a request/response pair is passed through the chain due to a
     * client request for a resource at the end of the chain. The FilterChain
     * passed in to this method allows the Filter to pass on the request and
     * response to the next entity in the chain.
     * <p>
     * A typical implementation of this method would follow the following
     * pattern:- <br>
     * 1. Examine the request<br>
     * 2. Optionally wrap the request object with a custom implementation to
     * filter content or headers for input filtering <br>
     * 3. Optionally wrap the response object with a custom implementation to
     * filter content or headers for output filtering <br>
     * 4. a) <strong>Either</strong> invoke the next entity in the chain using
     * the FilterChain object (<code>chain.doFilter()</code>), <br>
     * 4. b) <strong>or</strong> not pass on the request/response pair to the
     * next entity in the filter chain to block the request processing<br>
     * 5. Directly set headers on the response after invocation of the next
     * entity in the filter chain.
     *
     * @param request  The request to process
     * @param response The response associated with the request
     * @param chain    Provides access to the next filter in the chain for this
     *                 filter to pass the request and response to for further
     *                 processing
     * @throws IOException      if an I/O error occurs during this filter's
     *                          processing of the request
     * @throws ServletException if the processing fails for any other reason
     */
    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
        HttpServletRequest httpRequest = (HttpServletRequest) request;
        String uri = httpRequest.getRequestURI();
        if (uri.contains(FAVICON)) {
            // 过滤图标
            response.getWriter().write(CharSequenceUtil.EMPTY);
            return;
        }
        String contentType = httpRequest.getHeader(CONTENT_TYPE);
        if (Opt.ofNullable(contentType).filter(c -> c.contains(FORM_URLENCODED) || c.contains(APPLICATION_JSON)).isPresent()) {
            // ajax请求直接放行
            chain.doFilter(request, response);
            return;
        }
        // 判断请求是否来自手机
        String userAgent = httpRequest.getHeader(USER_AGENT);
        if (Opt.ofNullable(userAgent).filter(agent -> agent.contains(WINDOWS_NT)).filter(agent -> agent.contains(MSIE)).or(() -> Opt.ofNullable(userAgent).filter(agent -> !agent.contains(MACINTOSH))).filter(agent -> MOBILE_AGENT_LIST.parallelStream().anyMatch(agent::contains)).isPresent()) {
            request.getRequestDispatcher(MOBILE_PATH + uri).forward(request, response);
        } else {
            chain.doFilter(request, response);
        }
    }
}
